from python.extensions import db
from flask_login import UserMixin
from datetime import date

class WeatherData(db.Model):
    __tablename__ = "weather_data"  # 明确指定对应的数据表
    id = db.Column(db.Integer, primary_key=True)
    date = db.Column(db.Date)  # 对应表中的 date 字段
    max_temp = db.Column(db.Float)  # 对应 max_temp 字段
    min_temp = db.Column(db.Float)  # 对应 min_temp 字段
    weather = db.Column(db.String(50))  # 对应 weather 字段
    wind = db.Column(db.String(50))  # 对应 wind 字段
    aqi = db.Column(db.Integer)  # 对应 aqi 字段
    city = db.Column(db.String(50))  # 对应 city 字段，原代码里写的 urban 改为 city，和表字段一致
    aqi_level = db.Column(db.String(20))  # 对应 aqi_level 字段，原代码 air_quality_level 改为 aqi_level
    wind_level = db.Column(db.Integer)  # 对应 wind_level 字段

    def __repr__(self):
        return f'<WeatherData {self.city} {self.date}>'